import torch
from model_save import *
# 加载模型
import torchvision

model = torch.load('vgg16_mothod1.pth')
# print(model)

# 加载模型2
vgg16 = torchvision.models.vgg16(pretrained=False)
vgg16.load_state_dict(torch.load('vgg16_method2.pth'))
# model2 = torch.load('vgg16_method2.pth')
# print(vgg16)
model1 = torch.load('test.pth')
print(model1)
